package com.huawei.codearts.config;


import org.springframework.amqp.core.Queue;
import org.springframework.amqp.rabbit.config.SimpleRabbitListenerContainerFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

import java.util.HashMap;
import java.util.Map;

@Configuration
public class RabbitMQConfig {

    private RabbitTemplate rabbitTemplate;

    @Bean
    public Queue requestQueue() {
        /**
         * 配置直接创建队列
         * name 名称
         * durable 是否持久化
         * exclusive 是否排它， 如果有一个连接了其他就无法连接它，这个最好设置为false
         * autoDelete 是否可自动删除，没有绑定任何队列时删除
         */

//        Map<String, Object> args = new HashMap<String, Object>();
//        args.put("x-queue-mode", "lazy");  // 惰性队列 队列具备两种模式：default 和 lazy。
//        channel.queueDeclare("myqueue", false, false, false, args);

        return new Queue("requestQueue",true,false,false);
    }

    // 控制请求的并发处理数，可以通过配置并发消费者来实现。
    @Bean
    public SimpleRabbitListenerContainerFactory rabbitListenerContainerFactory(ConnectionFactory connectionFactory) {
        SimpleRabbitListenerContainerFactory factory = new SimpleRabbitListenerContainerFactory();
        factory.setConnectionFactory(connectionFactory);
        factory.setConcurrentConsumers(10); // 并发消费者数
        factory.setMaxConcurrentConsumers(20); // 最大并发消费者数
        return factory;
    }

    // 把RabbitMQ接收和发送的数据转为json
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    @Primary
    @Bean
    public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        this.rabbitTemplate = rabbitTemplate;
        rabbitTemplate.setMessageConverter(messageConverter());
        initRabbitTemplate();
        return rabbitTemplate;
    }

    /**
     * 定制RabbitTemplate
     * 1、服务收到消息就会回调
     *      1、spring.rabbitmq.publisher-confirms: true
     *         spring.rabbitmq.publisher-confirm-type: correlated
     *      2、设置确认回调
     * 2、消息正确抵达队列就会进行回调
     *      1、spring.rabbitmq.publisher-returns: true
     *         spring.rabbitmq.template.mandatory: true
     *      2、设置确认回调ReturnCallback
     *
     * 3、消费端确认(保证每个消息都被正确消费，此时才可以broker删除这个消息)
     *
     */
    // @PostConstruct  //MyRabbitConfig对象创建完成以后，执行这个方法
    public void initRabbitTemplate() {

        /**
         * 1、只要消息抵达Broker就ack=true
         * correlationData：当前消息的唯一关联数据(这个是消息的唯一id)
         * ack：消息是否成功收到
         * cause：失败的原因
         */
        //设置确认回调
        rabbitTemplate.setConfirmCallback((correlationData,ack,cause) -> {
            System.out.println("confirm...correlationData["+correlationData+"]==>ack:["+ack+"]==>cause:["+cause+"]");
            if (ack) {
                //如果投递到交换机成功
                System.out.println("消息投递成功!");
                System.out.println("cause1 = " + cause);
            } else {
                //投递到交换机失败，如果投递失败，可以做出处理，例如记录日志，发送短息等...
                System.out.println("消息投递失败!");
                System.out.println("cause2 = " + cause);
            }
        });

        //注意，在设置returnCallback的时候，必须要设置交换机处理失败消息的模式，设置为 true
        rabbitTemplate.setMandatory(true);

        /**
         * 注意：这个方法时消息没有抵达才会调用
         * 只要消息没有投递给指定的队列，就触发这个失败回调
         * message：投递失败的消息详细信息
         * replyCode：回复的状态码
         * replyText：回复的文本内容
         * exchange：当时这个消息发给哪个交换机
         * routingKey：当时这个消息用哪个路邮键
         */
        rabbitTemplate.setReturnsCallback(returned -> {
                    /*
                     * returned为回退的信息，包括以下几点：
                     * message   消息对象
                     * replyCode 错误码
                     * replyText 错误信息
                     * exchange  交换机
                     * routingKey 路由键
                     */
                    System.out.println("message = " + returned.getMessage());
                    System.out.println("exchange = " + returned.getExchange());
                    System.out.println("replyText = " + returned.getReplyText());
                    System.out.println("routingKey = " + returned.getRoutingKey());
                    System.out.println("replyCode = " + returned.getReplyCode());
                });
    }

}
